import rtconfig
Import('RTT_ROOT')
from building import *

# get current directory
cwd = GetCurrentDir()

src = Split("""
CMSIS/Device/ST/STM32G0xx/Source/Templates/system_stm32g0xx.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_cec.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_rcc.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_rcc_ex.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_flash.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_flash_ex.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_dma.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_dma_ex.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_pwr.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_pwr_ex.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_cortex.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal.c
STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_gpio.c
""")

if GetDepend(['RT_USING_SERIAL']):
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_uart.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_uart_ex.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_usart_ex.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_ll_usart.c']

if GetDepend(['RT_USING_I2C']):
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_i2c.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_ll_i2c.c']

if GetDepend(['RT_USING_SPI']):
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_spi.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_spi_ex.c'] 
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_ll_spi.c']

if GetDepend(['RT_USING_HWTIMER']) or GetDepend(['RT_USING_PWM']):
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_tim.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_tim_ex.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_ll_tim.c']

if GetDepend(['RT_USING_ADC']):
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_adc.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_adc_ex.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_ll_adc.c']

if GetDepend(['RT_USING_WDT']):
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_iwdg.c']
    src += ['STM32G0xx_HAL_Driver/Src/stm32g0xx_hal_wwdg.c']

path = [cwd + '/STM32G0xx_HAL_Driver/Inc', 
    cwd + '/CMSIS/Device/ST/STM32G0xx/Include',
    cwd + '/CMSIS/Include']

CPPDEFINES = ['USE_HAL_DRIVER']
group = DefineGroup('STM32_HAL', src, depend = [''], CPPPATH = path, CPPDEFINES = CPPDEFINES)

Return('group')
